package com.mh.controller;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.mh.dto.vo.BillStatisticsOfBookVo;
import com.mh.dto.vo.R;
import com.mh.pojo.AccountBook;
import com.mh.service.AccountBookService;
import lombok.AllArgsConstructor;
import lombok.NoArgsConstructor;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.annotations.Update;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.io.Serializable;
import java.util.List;
import java.util.Map;

/**
 * 账本表(AccountBook)表控制层
 *
 * @author zmh
 * @since 2025-02-08 09:18:48
 */
@RestController
@Slf4j
@CrossOrigin
@RequestMapping("/accountBooks")
public class AccountBookController {
    /**
     * 服务对象
     */
    @Autowired
    private AccountBookService accountBookService;

    /**
     * 获取首页账本统计数据（根据月份和账本ID获取指定月份的收入，支出和结余）
     * @param month 年月
     * @param bookId 账本ID
     * @return `
     */
    @GetMapping("/getMonthDataOfAccountBook/{month}/{bookId}")
    R<BillStatisticsOfBookVo> getMonthDataOfAccountBook(@PathVariable("month") String month,@PathVariable("bookId") Integer bookId){
        return accountBookService.getMonthDataOfAccountBook(month, bookId);
    }

    /**
     * 获取账本统计信息
     * @param bookIds 账本ID
     * @return ·
     */
    @PostMapping("/getBillStatisticsOfBook")
    R<Map<Integer, BillStatisticsOfBookVo>> getBillStatisticsOfBook(@RequestBody List<Integer> bookIds){
        return accountBookService.getBillStatisticsOfBook(bookIds);
    }

    /**
     * 分页查询
     *
     * @param page 查询页数
     * @param size 一页显示条数
     * @return ·
     */
    @GetMapping("/page")
    public R<Page<AccountBook>> getAllByPage(int page, int size) {
        //执行查询
        return accountBookService.getAllByPage(page, size);
    }

    /**
     * 通过用户id获取账本列表
     * @param openId 用户id
     * @return `
     */
    @GetMapping("/getAllByOpenId")
    public R<List<AccountBook>> getAllByUserId(String openId){
        return accountBookService.getAll(openId);
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public R<AccountBook> selectOne(@PathVariable Serializable id) {
        return R.success(this.accountBookService.getById(id));
    }

    /**
     * 新增数据
     *
     * @param accountBook 实体对象
     * @return 新增结果
     */
    @PostMapping
    public R<String> insert(@RequestBody AccountBook accountBook) {
        return R.success(this.accountBookService.save(accountBook) + "");
    }

    /**
     * 修改数据
     *
     * @param accountBook 实体对象
     * @return 修改结果
     */
    @PutMapping
    public R<String> update(@RequestBody AccountBook accountBook) {
        return R.success(this.accountBookService.updateById(accountBook) + "");
    }

    /**
     * 删除数据
     * @param userId 用户ID
     * @param bookId 账本ID
     * @return 删除结果
     */
    @DeleteMapping("/{userId}/{bookId}")
    public R<Boolean> delete(@PathVariable("userId") Integer userId, @PathVariable("bookId") Integer bookId) {
        return accountBookService.deleteById(userId,  bookId);
    }

    /**
     * 将账本设置默认
     * @param bookId 账本ID
     * @return ·
     */
    @PutMapping("/setDefault/{bookId}")
    R<Boolean> setDefault(@PathVariable("bookId") Integer bookId){
        return accountBookService.setDefault(bookId);
    }

    /**
     * 改账本名称
     * @param bookId 账本ID
     * @param bookName 账本名称
     * @return ·
     * */
    @PutMapping("/updateBookName/{bookId}/{bookName}")
    R<Boolean> updateBookName(@PathVariable("bookId") Integer bookId,@PathVariable("bookName") String bookName){
        return accountBookService.updateBookName(bookId, bookName);
    }

    /**
     * 新增账本
     * @param userId 用户ID
     * @param bookName 账本名称
     * @return ·
     */
    @PostMapping("/addBook/{userId}/{bookName}")
    R<Boolean> addBook(@PathVariable("userId") Integer userId, @PathVariable("bookName") String bookName){
        return accountBookService.addBook(userId, bookName);
    }
}

